# для ускорения построения графиков создадим новую функцию
def plotter(pltype, data, x, ax=None, y=None,
palette=None, with_darkest=True, hue=None,
style=None, title=None, subtitle=None, xlabel=None,
ylabel=None, no_legend=False, legend_out=True, legend_title=None, hue_order=None,
upper_lines=True, grid=True, color=None, figsize=(12, 6), **kwargs):
if ax is None:
fig, ax = plt.subplots(figsize=figsize)
if (color is None) and (palette is None):
palette = 'magma'
if (hue is not None) and (palette is None) and not with_darkest:
palette = sns.color_palette(palette, n_colors=data[hue].unique().size + 1)[1:]
elif not with_darkest and x is not None and y is not None:
palette = sns.color_palette(palette, n_colors=min(data[x].unique().size + 1, data[y].unique().size + 1))[1:]
plot_functions = {
'scatter': sns.scatterplot,
'box': sns.boxplot,
'bar': sns.barplot,
'hist': sns.histplot,
'line': sns.lineplot
}
plot_functions[pltype](data, ax=ax, x=x, y=y, palette=palette, hue=hue, hue_order=hue_order, color=color, **kwargs)
if pltype not in plot_functions:
raise ValueError('Неизвестный тип графика')
if hue is not None and legend_out:
sns.move_legend(ax, bbox_to_anchor=(1, 1), loc='upper left', title=legend_title)
ax.set_title(title)
if subtitle is not None:
ax.text(y=1.05, x=0, transform=ax.transAxes, s=subtitle, fontsize=12)
if xlabel is not None:
ax.set_xlabel(xlabel, loc='left')
if ylabel is not None:
ax.set_ylabel(ylabel)
if upper_lines:
ax.plot([0.05, 0.9], [0.99, 0.99], transform=fig.transFigure, clip_on=False, color='black', linewidth=0.7)
ax.plot([0.05, 0.9], [0.90, 0.90], transform=fig.transFigure, clip_on=False, color='black', linewidth=0.7)
if no_legend:
ax.get_legend().remove()
if grid:
ax.grid(alpha=0.3, zorder=1)
return ax
